run train-1.ipynb to generate results from paper 